import json
import pandas as pd

# 定义任务类别和初始提示词
task_prompts = {
    "text_generator":[
        "请生成一段关于人工智能的介绍",
        "用简洁的语言描述机器学习的基本概念",
        "编写一段500字的短文，主要是绿色能源"
    ],
    "translation":[
        "将以下文本从中文翻译为英文:{text}",
        "将下面的句子从法语翻译为德语:{text}",
        "翻译下面内容为日语:{text}"
    ]
}

#保存为json文件
def save_prompt_library(prompts,file_name='prompt_library.json'):
    with open(file_name,'w',encoding='utf-8') as f:
        json.dump(prompts,f,ensure_ascii=False,indent=4)
    print(f"提示词保存成功{file_name}")

save_prompt_library(task_prompts)